/*
 * Copyright (c) 2003-2010, Live Media Pty. Ltd.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification, are permitted
 * provided that the following conditions are met:
 *
 *  1. Redistributions of source code must retain the above copyright notice, this list of
 *     conditions and the following disclaimer.
 *  2. Redistributions in binary form must reproduce the above copyright notice, this list of
 *     conditions and the following disclaimer in the documentation and/or other materials
 *     provided with the distribution.
 *  3. Neither the name of Live Media nor the names of its contributors may be used to endorse
 *     or promote products derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package com.colony.transport.socket;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Stack;

import com.argot.remote.transport.TypeEndPoint;
import com.colony.transport.Client;

public class SocketClient
implements Client
{
	private static final int MAGIC1 = 10;
	private static final int MAGIC2 = 20;
	
	private int _port;
	private InetAddress _host;
	private Stack _stack;

	public SocketClient( String host, int port ) 
	throws UnknownHostException
	{
		_host = InetAddress.getByName(host);
		_port = port;
		_stack = new Stack();
	}

	public TypeEndPoint openLink()
	throws IOException
	{
		synchronized(_stack)
		{
			if (!_stack.isEmpty())
			{
				SocketEndPoint ep = (SocketEndPoint) _stack.pop();
				if (!ep.getSocket().isClosed())
				{
					ep.getOutputStream().write(MAGIC1);
					ep.getOutputStream().write(MAGIC2);
					return ep;
				}
				System.out.println("socket closed!");
			}
		}
     	Socket socket = new Socket();
     	InetSocketAddress addr = new InetSocketAddress( _host, _port );
     	socket.connect(addr,3000);
		socket.getOutputStream().write(MAGIC1);
		socket.getOutputStream().write(MAGIC2);
		return new SocketEndPoint( socket );		
	}
	
	public void closeLink( TypeEndPoint endPoint )
	{
		synchronized(_stack)
		{
			_stack.push(endPoint);
		}
	}

	private static class SocketEndPoint
	implements TypeEndPoint
	{
		private Socket _socket;
		private InputStream _in;
		private OutputStream _out;
		
		public SocketEndPoint(Socket socket)
		throws IOException
		{
			_socket = socket;
			_in = new BufferedInputStream(socket.getInputStream());
			_out = new BufferedOutputStream(socket.getOutputStream());
		}
		
		public InputStream getInputStream()
		{
			return _in;
		}

		public OutputStream getOutputStream()
		{
			return _out;
		}
		
		public Socket getSocket()
		{
			return _socket;
		}
		
	}
	
}
